# 这里是用于展示的图像。就是输入一个数字，然后可以直接展示出来你想要展示的那个数字。
import cganclass_function as mycf
import torch

# device = 'cuda' if torch.cuda.is_available() else 'cpu'

noise = torch.randn([1, 100, 1, 1])  # 这里是生成噪声

print('Please enter the number you want to get')
num = input('Your number is : ')  # input输入的是一个字符串，这个要记得强制类型转化。
label = torch.tensor([int(num)])
mycf.show_one_image(noise, label)
